package com.hk.common.websocket;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

/**
 * @author zhuiyv
 */
@Configuration
@EnableWebSocket
public class WebSocketConfig  implements WebSocketConfigurer {

	/**
	 * 前端请求路径配置类
	 */
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
    	
    	
    	//添加handler
    	registry.addHandler(webSocketHandler(), "/websocket").setAllowedOrigins("*").addInterceptors(new WebsocketInterceptor());
    }
    
    @Bean
    public WebSocketHandler webSocketHandler() {
        return new WebsocketHandler();
    }

}